Clean up and replace method in memento