Script update to handle ci build