Demo of Numba solving the wave equation in 2D