So I just got my FRS back from Toyota, the car was having a hesitation issue at 4000rpm and sputtery misfirey cold starts that'd seem to level out after reaching closed loop. they replaced the high pressure fuel pump which seems to have solved half of my issue. Now I'm getting 2 air fuel imbalance codes.
The sputtering cold starts sounds exactly like this:
https://youtube.com/shorts/8vZuumwxIKw?feature=share
Any suggestions are greatly appreciated and if you have experience with this issue your input could go a very long way.