Minor fixes + updated docs.