Add barebones AI mode in