added functionality to load dataset, requires ~15 sec