Updating the refactoring demo to reflect the changes.