Training and what is too much at once

Hi all me again with the 11 week cocker spaniel

Thank you all so much for the help and advice recently, hes at 7.5 hours in crate with 0 accidents overnight.

Now i would like to ask/get advice with training. He know sit, lie down, hi 5 (paw) and done well with wait (not fully there yet say 50%), off, enough (for biting feets, teething landshark yay)

But he doesnt do well as of now with return (responds to name somewhat, but is easily distracted) and retrieve (loves getting ball, but doesnt want to return ball as an example) and impulse control.

Any tips to help with training him, its hard to see if its all too much at once.

Ive got puppy classes booked in 3 weeks for him (Timings didnt work out or would have been sooner)

Il stress i know hes still a puppy and isnt going to be perfect.

Thank you all again!