Update scale.py