adding more ui stuff