Small changes in custom script file