Added edit and delete function