Fixed it. Added undo.
improved encapsulation
back button