made some improvements to the UI