fix last one in hw3
update