Added rescale_square