Updated Assignment 2 file