Continue development of UI class