updating script a bit