Add unit in comment