Fixed the x direction bug, was actually an issue with attempted y axis flipping few commits ago. Also changed scene to better show the diffusion results