add constrain_weights
change sigma_x
update_params now return 3 values