User Input Array Size Python Interview