Fixed a couple of code comments