work from week4's class
Finished Homework 4