Modified example to have Neumann BC