Changed script a little