That was the idea I was going for, I really enjoyed that game too.
The AI can absolutely be improved. I was starting to overthink the whole AI and I have a bunch of useless checks in it now but if I keep working on it, it can become good I think. I've finished a more polished one (visually that is) which I'll put up once the rating is over.