added Dataset class