Replaced checkbox with toggle of edit button