Add function to load preprocessed dataset