small changes in hw3.3