Electors

import java.util.*;

public class ElectionsTester {

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        //Map<Integer, String> electors = new TreeMap<Integer, String>();
        Set<Voter> voters = new HashSet<Voter>();
        boolean flag = true;
        char ch;
        do {
            System.out.println("Press 'N' to enter a new elector, or 'Q' to quit");
            ch =sc.next().charAt(0);
            sc.nextLine();
            if(ch == 'Q' || ch == 'q') {
                flag = false;
            } else if (ch == 'N' || ch == 'n'){
                System.out.println("Please enter your name: ");
                String name = sc.nextLine();
                System.out.println("Please enter your ID number: ");
                int id = sc.nextInt();
                Voter v = new Voter(name, id);
                if(!voters.add(v)) {
                    System.out.println("Voter already exists!");
                }

//                if(electors.containsKey(new Integer(id))) {
//                    System.out.println("Elector already registered!!!");
//                } else {
//                    electors.put(new Integer(id), name);
//                }
            } else {
                System.out.println("Bad input!!!");
            }
        }while(flag);
        System.out.println(voters);
//        System.out.println(electors);
    }
}