Update Problem #4.py

updated docstring