Added code for loading datasets