starting to re-architect so that any generic dataset can be used, audio_dataset.py and prepare_datase2.py should be useful for this