First Tests for diffusion energy function added