Modify setup script