Working on edit function, added modal