Update script to support the new configuration