Add solution.py