Update assignment 1