2020 Audi Q5 E

My Transmission failed to go in forward and was racing in drive. I put it reverse and that worked but put back into drive it races then makes a bang and instantly goes into drive. I had car towed to Audi of Allentown. They initially went over vehicle couldn't find issue then it did it to them. They said it had a code for solenoid. They changed valve body and it had codes so they were in contact with Audi Hotline they had them change Mechatronic unit also and the issue was solved, They sent me video, I picked up car and gave them loaner back, I used car local after a few times then it did it again. Had car towed to Audi of Allentown again, they couldn't find issue and sent me video of car. I went to pickup vehicle and they were getting vehicle ready and it did it to them and they said they would let service know. They gave me a loaner. I've called and they haven't had vehicle do it again until today. They videoed it and sent it to Audi Hotline (engineers). There waiting on a reply from Audi. The vehicle is unsafe-- if it did it to me while entering a highway and cut out I could be hit in from rear and it would be a catastrophic accident. Or if it did this at a crosswalk that could be another issue but with pedestrians. If I get car back and it happens again then it cant be fixed and its unsafe I'll need to get a lawyer to rectify this situation.