Move setup.py file into docs/
Minor mods to notebooks