Modified ui.py so it will run locally