fix up mu -- now does pointwise divide (oops)
fix dimensions on grad desc
re-add constant term