Fix tests, update docs.