minor fixes in hw4