updated ui stuff for lab 4