Looking at Memento

Changed the implementation to match what's in the book.