Add check for diffusion units into initialization checks

Add check for diffusion units into initialization checks #371