improved UI design