Added class comments
Improved description in README file