Working on UI class