switched to regression model