Simplifying problem to fewer grid points. Looks like both implementations give the same results.