Update working version 1.py