fixed comments
ass2 done