added more UI elements