Add proper docstrings to all class methods