Create stiffness_matrix.py