Array In Python User Input Function