Updated the file based on class feedback