Update input_dialog.py