add user input in Python