Add rescale_array function