Fix docs

I'm working on creating a draft of the documentation based on MDTraj's documentation framework.