Added additional shape functions, fixed Diffusion (somewhat.)
Using fields that are too small cause problems for the solver.