Update docs

AttributeError: 'Dataset' object has no attribute 'extend'