added AI option - still rough