Adding a parameter to a decorator