Fixing the docstring. Had wrong information about the returned map.