comments update

Placing comments at sections where you can clean your code up a little bit if it's possible.