the only one acceptable way to discipline a puppy, is to behave exactly like mother dog - shake him by the scruff a little (not lifting his full weight of course) and bring to his bed/sleeping mat/crate.
but distracting him with toys is so much healthier for the animal. and you get to play with a cute puppy as well 🙂