Modify publish script