Python 1: Lesson 16: Refactoring Example - added new functionality