Script cleanup
Small fix
Merge pull request #292 from ninjalulz/master

nice work :D