Update problem2.py

fixed spacing