Array menampilkan nilai dari sejumlah siswa dimana jumlah siswa

Buat program untuk menampilkan nilai dari sejumlah siswa dimana jumlah siswa tergantung dari masukkan user,dan semua nilai diinputkan, kemudian cari :
    a. Jumlah
    b. rata-rata
jawab :
import java.util.Scanner;

public class Rata
{
    public static void main(String [] args)
    {
    int total=0;
    float rata;

    Scanner scan=new Scanner(System.in) ;
    System.out.print("Berapa kali memasukkan nilai = ");
    int nilai=scan.nextInt();

    int arrayNilai[]=new int[nilai];
    for (int i=0;i<nilai;i++)
    {
        System.out.print("Nilai ke-"+i+" = ");
        arrayNilai[i]=scan.nextInt();
    }

    for (int j=0;j<nilai;j++)
    {
        total=total+arrayNilai[j];
    }
    rata=total/nilai;
    System.out.print("Total Nilai = "+total);
    System.out.print("\nRata-rata Nilai = "+rata);
    }
}