Revert "Update update.py"

This reverts commit 51e5f67d05399cd7156f5714b060f980a6af4001.
Revert "Revert "Update update.py""

This reverts commit 3e5b5f1faa32d27b72ad343a7b4756fbdee1f21a.