Allow passing additional arguments into the callback