Refactored the dataset.

Signed-off-by: Xiaomao Chen <xiaomao5@live.com>