Added code to visualize subset of dataset.
Merge remote-tracking branch 'origin/master'