Create problema1.py

Así se ve mucho mejor.