Array In Python User Input