Undo functionality and moar buttons