made some correction into the docstring of the model