put the code into __init__.py