Modified setup script.