Adds undo button. Fixes #31