Edited TODOs in main.py.