Update to code comment