Adding solution.py file