fix hw4 feedback!