Update UI.py

Some little changes, which clean code.