Fixed beta update - now covariates can be included in model