UI done. Only need to add icons!