Got edit button working.