added code to train on eeg feats