Improve visual ui