Continuing work on AI

Signed-off-by: James Marsh <james.marsh124@gmail.com>