Edit gen_data.py and sdp_solver.m