Update to code comments