Update assignment 2