Should these just be moved to parameters to `__init__` ?