Add AI! (mostly framework :/)