Fixed tests and updated docs.