Following http://lorenabarba.com/blog/cfd-python-12-steps-to-navier-stokes/