Moved update script into function.
Merge branch 'release/2.0.1'