Assignment 2: Diffusion