Added dtype argument in docstring