Updated code comments

Finally!