Add tests for distinctBy