I want to implement diagonal penalty method for enforcing the Dirichlet boundary conditions. That is, the diagonal entries corresponding to Dirichlet boundary are going to be scaled by a large number. What is the easiest way to do this? Thanks!