Made the setup script more user-friendly