You can work on developing a "2 On 2 Off" (2o2o) for contacts fairly easy.
Start with "Target" practice. Get the lid of a small container, like a butter tub or cream cheese container. (Make sure it is clean.) Put it on the floor with a treat on it and when the put picks up the treat, reinforce with verbal praise (yes) or a click. Work up to where you can have the pup touch the target with no treat on it for the click/treat or verbal praise/treat. Put the word "Target" or "Spot" (or something similar) to this object as you work on it.
As they get the hang of "Target", get a wood plank, it doesn't have to be standard length/width but it helps, and lay it flat on the floor. Place your Target (baited with a treat) at the end, on the floor. Have Kai walk across it and stop at the end with two 2 front paws on the ground and rear 2 on the plank and take the treat from the target; click or verbal praise and treat. Slowly remote the treat and just leave the target - they get the treat when they touch the unbaited target with 2o2o.
Another exercise you can do is similar to cavaletties (a tool used in equestrian and agility training to help with stride and learning where their back legs are). All you need for this is a standard ladder. Lay the ladder on the ground and walk your pup through it. The idea is for them to walk through the ladder without hitting the rungs. As usual praise and reward for even the smallest progress.
Above all - MAKE THESE FUN GAMES for the pup. Don't work on them for more than a few minutes at a time.