I think this would be even better placed in the `__init__` block