script improved
correction bug