adjusted 2d plot in test_regression added in last commit