Improved MLLib example, improved data set