Modified instance creation logic. Setting the history dataSet to the test data