Fix diffusion boolean and make it class variable