fix hw4_2