Add missing files for script