Leer numero y elevarlo al cuadrado o al cubo

import java.util.*;
public class cudro{
    public static void main(){
        int x, num;
        Scanner teclado = new Scanner(System.in);
        System.out.println("Escribe el numero");
        num = teclado.nextInt();
        System.out.println("1.Elevar al cuadrado \n2.Elevar al cubo");
        x = teclado.nextInt();
        if (x ==1){
            System.out.println(num+" al cuadrado es: "+num*num);
        }
        if (x ==2){
            System.out.println(num+" al cubo es: "+num*num*num);
        }
    }
}