Added some basic AI