Clean up imports in EdgeRDD