Improve UI and data visualization