I think rather than finding the `NETWORK` element in each of the methods, we should define a variable in the `__init__` function and then use it in the methods.