adding some more ui stuff