final changes for hw4