Lesson04 Homework

I created a new implementation of this in Lesson 5 folder. Please check that folder for the latest code