class Solution { public int solution(int a, int b, boolean flag) { if(flag == true){ return a + b; } else { return a - b; } } }