Updated the files for assignment 2