Added some missing comments to the code