making final changes for Hw4