Fix docstring on classes