Changed some comments to look better inline with the code