Here, you might want to add `@functools.wraps` on the `decorator` function.