Moved ai to constructor so we have one less line new-ing up each time