Add icons, fix minor bugs