[python] Getting user input