Use pytest - added __init__.py