Added main.py and created functions