made dataset inherit from  base dataset class
prep data save path is now cleaner