Update. I towed the car to another dealer in another city that has the tool. The cleared the transmission codes and it shifts like new. No remapping needed. Just a clearing of the codes. Which they basically did for free since they included it in the labor for replacing my belt and idler pulley.