Add initial dataset retreival