Update ui.cpp