Create main.py

Main function.