Update entrainment.f90