Rename docstring return classes