Added more ui