Updated imports in main.py and added example.