Change comments to better match documentation