small changes

added some comments in code