how to get user input for list in python