Revert "Revert "Update update.py""

This reverts commit 3e5b5f1faa32d27b72ad343a7b4756fbdee1f21a.