Added future work note in __init__.py file