Fixed docstring class references