Added comments
Tiny fixes