Fixing comments regarding units