Add production stage