برنامج بلغة ال Java لجمع أو طرح قيمة وزنين مدخلين من قبل المستخدم

import java.util.Scanner;
import java.text.DecimalFormat;

public class Weight {
	public static void main(String[] args) {	
		Scanner input = new Scanner(System.in);
		double a,s;
		DecimalFormat e = new DecimalFormat("0.00 kg");
	    System.out.println("Enter A Weight 1");
	    double  w1 = input.nextDouble();	
	    System.out.println("Enter A Weight 2");
	    double  w2 = input.nextDouble();		
	    System.out.println("enter 1 to chose add or 2 to chose sub");	    
	    int h=input.nextInt();
	    if(h == 1){
	    	a = w1+w2;
	    	 System.out.println(e.format(a));
	    }
	    else if(h == 2){
	    	s = w1-w2;
	    	 System.out.println(e.format(s));
	    }
	}
}