added non-linear correlation