Here you will find useful jupyter notebooks on this subject.
Choose your favorite user interface:
Diffusion equation: