Added button and event listener for edit