Update ui.py

updated