Fix diffusion term
Fixing python loop head printer