Added tracking of performance in validation set