Add documentation for return_distances option