(DIRTY COMMIT) working on the punishment module