Python User Input Array Size