Homework 4.4 done and tested