Update assignment 5