Add test to check delete functionality