Added a bit of AI
-