Fix docstring
Make extension more descript