make changes for hw4