Python User Input Array C