Need to work on the modal function for edit