fixup! Added dataset 2 to the indexed dataset generator code