Update gradient descent in linear regression
Add documentation to functions
Add plots