Revise paste Main.java

import java.util.ArrayList;
import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        ArrayList<Mahasiswa> mhsList = new ArrayList<>();
        int pilihan;

do {
            System.out.println("Menu:");
            System.out.println("1. Tambah Mahasiswa");
            System.out.println("2. Tampilkan Mahasiswa");
            System.out.println("3. Keluar");
            System.out.print("Pilih menu (1-3): ");
            pilihan = scanner.nextInt();
            scanner.nextLine(); // membersihkan newline

switch (pilihan) {
                case 1:
                    System.out.print("Masukkan NIM: ");
                    String nim = scanner.nextLine();
                    System.out.print("Masukkan nama: ");
                    String nama = scanner.nextLine();
                    System.out.print("Masukkan umur: ");
                    int umur = scanner.nextInt();
                    
                    Mahasiswa mhs = new Mahasiswa();
                    mhs.setNIM(nim);    
                    mhs.setNama(nama);
                    mhs.setUmur(umur);

mhsList.add(mhs);
                    System.out.println("Mahasiswa berhasil ditambahkan!");
                    break;
                case 2:
                    System.out.println("Daftar Mahasiswa:");
                    for (Mahasiswa m : mhsList) {
                        System.out.println("NIM: " + m.getNIM() 
                        + " Nama: " + m.getNama() 
                        + " Umur: " + m.getUmur());
                    }
                    break;
                case 3:
                    System.out.println("Keluar dari program.");
                    break;
                default:
                    System.out.println("Pilihan tidak valid. Silakan coba lagi.");
            }
        } while (pilihan != 3);

}
}