made it so, that model keyword for correctDrift, now may be a string, or a callable
model
correctDrift