small changes

added some comments