Fix bug, improve docs.