FIX Improved the update script