python code: finding analytic solutions for height and velocity: plotting the solutions and model output