Added homework 4, fix for what I got wrong in hw3