new file:   solution.py