Minor update to code comments