time step optimisation

Mass balance and flux still in progress