Определить входит ли цифра в данное число

public static void main(String[] args) {
    Scanner in = new Scanner(System.in);
    System.out.print("Input number: ");
    int number = in.nextInt();
    System.out.print("Input digit: ");
    int digit = in.nextInt();
    while(number > 0){
        int part = number % 10;
        if(part == digit){
            System.out.println("YES");
            return;
        }
        number = (number - number % 10) / 10;
    }
    System.out.println("NO");
}

public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        long number;
        int digit;
        System.out.print("Number: ");
        number = scan.nextLong();
        System.out.print("Digit: ");
        digit = scan.nextInt();
        boolean flag = true;
 
        while (number > 0) {
            if (number % 10 == digit) {
                System.out.println("True");
                flag = false;
                break;
            }
            number /= 10;
        }
 
        if (flag) System.out.println("False");
 
    }