Added basics to main.py