fixed script
fixed script more
Merge pull request #2 from Arctangent1759/master

Fixed script more.