Applied luborliu/patch-3

Improve the efficiency of the code
New Features: Predict which clusters fit a RDD