added the ability to use different datasets