adding the new punishment
forgot to import
derp