update for assignment 3.4