Update input.py

Didn't I test this?