update for assignment2