added an ipython notebook to figure things out
working version

It’s now fully functional