More work on PDE implementation