Datahandler returns training & test data now.