Added methods to AI