Fixing up code comments