added some more ui stuff