realized it can be done in __init__.py :)