Add edit button to template