Add production image support