Added support for wikipedia dataset