Implementing the newly created edit button