changed compute_psd_epochs to returngenerator=false
added save lines back into script