Added documentation to function
Added correct dtypes to parameters in documentation