Added undo action! :D

Yeah it works fine, need to add button though