Add setup script
Fix the setup script