small improvement in script