User Input With Classes In Python