Worked on HW6

Linear gradient for lasso has error. Fix!