Remove pdb.set_trace()