Better docs, minor changes