Cleanup
Working on docs and setup.py