Added edit and delete functionality back