Add initial AI code