Some comments on the code

Also removed tab and made it 4 space. Hope that's ok