You could try giving him a treat once he unballs

This will encourage unballing which means he will do it faster once he catches on that he gets a nommy treat once he shows his cute lil face ^-^
Seems like you're doing a good job though
You just gotta give him time