Added test

I got tired of loading up ipython notebook and getting limited feedback,
so I reimplemented the demo code as a normal python script. Also, fixed
up some argument problems to the various kernels.