Update setup script for production