Update calculate_flux.f90