Update problem1.py

Fixed spacing