Prepping for adding AI