Adding more functionality to edit