eliminated z-drection in mu, flux, etc variables for consistency in 2D