Initial version of main.py