Update assignment 3