Linear Regression: Clean up output and documentation