Added method to extract a sampled subset
Refactored colorization, eliminated rdd option