Create cost_matrix.py