added basic regression with viz