Create main.py

main driver