Started adding step 5 - 2D linear convection.