Added remove punishment method added