Update assignment 4