Added main methods and classes to YoloAI