Add a function to plot the mesh