Update/edit is finally working