fix a bug of problem3
hw4