Third argument changed to unity_gain_flux
(#552)
Minor changes
(#552)