Fix math in docstrings and fix docstring style to conform to google style in registration.