Recommendations on improving the AI algorithm?