Code revision
adding comments to program