Updated SolveForKMatrix.py to print K matrix, still buggy