Update publish and stages scripts