added dataset file locations and changed save process