added dataset helper class