Added class Base for dataset creation