expanded and revised sample_problems.py