did simple refactoring exercise
minor change to simple refactoring exercise