Kindly check this code

Couple of mistakes:

  1. Don’t use stoi as overflow may occur, direct string comparison is okay.
  2. Print endl after every test output.