Create solution.py