Update assignment 6