Finished Lesson 5 Homework Assignment, although could use some tidying