How to take a user input for list in python