user input in python - Codanics