Change parameter passed to function

Change parameter passed to scale_array() to correct parameter
Change function return type

Function create data_sets() now returns a tuple containing the training
and test set
Fix downsample return type