This sort of attitude is a big reason why US politics is polarized around this issue. So let me try to adjust your "small punishment for a greater good" attitude a bit.
Guns are self-defense weapons. They are particularly useful as equalizers: a little old lady who can shoot can defend herself against a mugger much larger than her in a way she couldn't possibly do by any other means. So any government that proposes to take away people's guns "for a greater good" has an absolute obligation to extirpate any kind of violent crime that might cause a person to need a gun for self-defense. I mean "extirpate" literally: zero violent crime. Or at least so close to it as to be rounding error. But of course no government on Earth ever has or ever can achieve that level of control over violent crime, at least not in a free country. Perhaps China could do it, though even there I doubt it: China has criminal gangs.
So given that the government can't possibly make it truly unnecessary for any law-abiding citizen to need a gun, if the government still insists on taking away people's guns "for a greater good", the government is taking the attitude that, as long as the politicians who live in gated communities and have personal protective details can have their bodyguards have guns so they aren't victims of violent crime, they don't care that the people they're supposed to be representing are victims of violent crime and can't defend themselves. And that is not an acceptable attitude for the government of a free country.