Fix Undo/Redo Bug

Main problems were: accepting specific states and executing those states properly. Tested this for a while, but pushing this out ASAP as the current undo/redo is pretty damn buggy.

Fixes Issue: #2