Changed the dataset conversion code slightly