moved dataset load to own classes