Add tests for new remove all functionality