fixed bugs in dataset