Initial work on main.py