Fixing up comments