add ui for lab 2 part 2