working on undo

undo working, animation is funky