Remove pdb.set_trace