Adjusting update script for extended API