Lesson 01 - Added cleanup code