working on edit feature