Update test_gui.py