nit: We normally put the docstring on the type and not on its implementation. mind moving it? :)