add icons, improve UI