Add save function for evaluation