Added transport equation visualisation