Adding in icons and more ui prototypes