Add tests for drop