Add comments to code so it is understandable