Include ability to change psd when loading data

Modify the PSD when loading in data so that the model trains on randomised PSDs