Add ground to plot

We want to improve the visualisation by showing the ground. The ground function needs to be stored in the simulation output data.