Added the hidden representations for training set