Updated scirpt