working on edit button