Try using validation set