Updating example due to refactoring