Final improvements to weight and regularization models, and notebook explaining them