Updated code comments