Add drop function
Add test case