Trajectory prediction for AVs in traffic