add the option to load saved file in kaggle small dataset