Do we need to add these two lines in example of each function? I think this should only be in the `__init__` function.