New version using mu and a max iteration count. Works for the toy data with these values.