Test Set validation added, as well as MSE