added todo in main.py