Undo functio need implementing.

Need to find new way to implement undo.