Add test function.
Add test function
Docstring fixes
Small docstring improvements.