little changes
just added some comments