updated ui.py