Python Array Input From User