Updated assignment file with code and comments