work on edit mode