Pinpointed where code needs to go, added comments

I added some comments kinda explaining the problem and where we need the
extra code.