Create solution4.py