Slight update to setup script.