Changed edit button to icon