Update Linear Regression chapter figure code, round 1