Updating doc strings and minor test change.