Reaction Diffusion solver in Python