changed script a littlebit