Hook up edit button