Python User Input Array