Working on create-input.py