DOC improve docstring about parameter type