MasterThesis/ diffusion.py