- Implemented unit undo functionality