Update ui_module.py