added function to improve regressions