Updated to include Poisson regression