Re-done EDIT button colour-change