Creating penalty add/edit logic