Create script.py

Solution in Python