Update __init__.py

Fixed business rules logic