Implemented function to load dataset