Please use a single argument accepting a callable. This way, any callable will be supported (the code already supports all types of callables internally)