Add UI elements and functionality