Updated script.py

Cleaned up code