Refactor MLlibIntegrationSuite