+ Add 20 news groups into the data sets.
+ Refactor the dataset.py a little bit.