Make edit button change color when editing enabled