Update __init__.py

Fine-tuned file matching