How To Get User Input For Arraylist In Python