Updated assignment2 files