Equation updates

Ready to derive diffusion