Implement edit button