Fixes to programming assignment 2

I added comments