How can I fix the automatic transmission on an Audi A2?

I recently bought a used Audi A2 and have been experiencing issues with the automatic transmission. It seems to be shifting roughly and sometimes gets stuck in gear. Any advice on how to fix this issue?

First, check the transmission fluid level and quality. Low or dirty fluid can cause shifting problems and should be changed or topped off if necessary. Also, consider getting a transmission flush to remove any buildup. If the issue persists, it may be a mechanical problem and you should consult a mechanic.