Added docstring to class