Added subsampling for training the network, checking results now