Clean up docstrings for math module.