Style the edit button