Assignment 4 DiffusionSolve