fix some bug hw day 14

fix some bug hw day 14