Updated soltuion.py test.py