added separate module for AI code