Update karplus_strong_mat.py