Store User Input Python