fixed some problem in hw week 5