Learning how to punish