adding in hw 4