more problems.py solutions