zzz fixed the error for hw4p1