How to take user input in Python