Create main.py

Código original.