Add commented out main

In case you want to run it outside ipython