[WIP] My edits on Jason's ipynb

I produce very different joint angle results. So I think I introduced another bug. I do have a different version of scipy, so that could also be the cause. I will check this.