Update assignment 2 files