Using Input Function In Python Assignment