First, it should have an option to just play again, instead of having to go back to the main menu every time.
Second, have an option to be X or O and who goes first (and/or random).
Third, it has a big problem - if IA places an O to block you, you can just put the X on top of it. Basically, you can put the X in ANY box, whether it's open or not!