Edit function and Add/Edit prompt added