Adjust setup script