write docs
Draft for docs (set up read the docs)