Added boundary conditions. Not sure what the best way to do Dirichlet boundary
conditions is.