hw4 commented and edited