This is great! You can make an ipython notebook version of this too.