updated notebook with test case to see if time step affect diffusion