Improved visual design, added undo/redo buttons