Add production flag to stages