update for hw 4