set the comments to the code

I've forgotten to  do it earlier...