Trainer for AI added. Need to make a network and train it here.