Fixing comment
Fixing minor comments