Added content to init.py file