small changes in script