edit button working