Added in icons
Implemented the undo class