The second idea is spot on and it's something that I sometimes struggle to discern, especially with adults. In the long term, punishing bad behaviour doesn't really do much since the underlying problem isn't addressed and solved.