Also the position of the new argument should be later, so that exisiting scripts relying on positional arguments won't be broken.
I would suggest adding a keyword argument that defaults to None and handle the cases in the function body.