well done FancyPantsJen
You can apply the same kind of method to anything your puppy does that you want to change. Just think about what you are doing to "reward" the behaviour eg paying him attention, including yelling at him, picking him up or chasing him, and find another way to get what you want.
So glad it has been easy for you to turn around. Don't be disappointed if you have the occasional set back - just step back to an easier level of expecation and start again.