Update docs, minor fixes