Bathroom ceiling fan wiring help

Hello,

I am replacing an old bathroom ventilation fan and am looking for some help with the wiring. The old fan and light was controlled by one switch. The new fan, because my wife is expensive and irrational, has a bluetooth speaker and nightlight.

It seems to me like I will need to run another wire which would connect to the black from the fan and the black(switch connection) from the switch. The yellows would connect to the yellows and the red to the black(power) from the outlet.

Can someone please confirm if this is correct or correct me?

Thanks!