slight modification to setup script