Update code comments

Update code comments