Got checkmark box working only on edit page