plt.legend(loc="upper left",
                   labels=[y_label1, y_label2],
                   bbox_to_anchor=(0, 1.17))
        # remove lines marking units on the axis
        ax.xaxis.set_ticks_position('none')
        ax.yaxis.set_ticks_position('none')
        plt.show()
        plt.close()


data = pd.read_csv("cleanedEconFreedomData.csv")
reg = Regression()

y_var = ["GDP per Capita (PPP)"]
x_vars = [
    "Judical Effectiveness", "Property Rights", "Fiscal Health",
    "Investment Freedom ", "Financial Freedom", "Inflation (%)",
    "Public Debt (% of GDP)"
]
reg.OLS("GDP Per Capita Restricted", data, y_var, x_vars)
plot_scatter_with_estimator(reg.data, x_vars, y_var)

x_vars_unrestricted = x_vars
x_vars_restricted = ["Judical Effectiveness", "Property Rights"]
reg.OLS("GDP Per Capita Restricted", data, y_var, x_vars_restricted)
reg.OLS("GDP Per Capita Unrestricted", data, y_var, x_vars_unrestricted)

joint_f_test = reg.joint_f_test("GDP Per Capita Unrestricted",
                                "GDP Per Capita Restricted")
joint_f_test.to_csv("Joint F_test; y = " + reg.y_name[0] + "; " +\
                    joint_f_test.index.name + ".csv")