Enhanced setup script.