Add stencil code for HW 7