Add docstring and extra test for localization