Added Punishment Classes

Will work on later.