Please can someone check that I have understood this part of the code for my game or advise a better way of showing this?
//--------------------------------
// DIFFICULTY
//--------------------------------
void Difficulty()
{
Top_Border();
cout << "\n Choose computer difficulty level.\n"
" (E)asy\n"
" (A)verage\n"
" (H)ard\n\n"
" ? ";
cin >> ans;
if(ans == 'E' || ans == 'e') { diff = 1; }
else if(ans == 'A' || ans == 'a') { diff = 2; }
else if(ans == 'H' || ans == 'h') { diff = 3; }
else Difficulty();
}