Updating Mult + Logistic Regression chapter figure code