Remove unused parameters, update docstring