dataset creation class