This line is misleading -- Can you delete it ? We seem to set the first two features to 10 always if I am reading things right.. Why don't we have two versions of this function ? One where we take in the feature values as Array[Double] for the test and the other for the generator where we pick randomly ?