Added correct code to refactored example.