Implement testing on validation set