Added changes to SpectralTrainClass