Minor updates to linear regression plotting functions