Update cleanup_input.py