Fix to code comments