Taking User Input For Array In Python