Update for new script