minor fixes in hw 4 and 5