Adding FEM code