Add UI logic