Update intermediate001-solution.py