Working on the UI class