Updated and tested the delete functionality as well