Added documentation in the class