Update

Uncommented the lines to setup the dataset.