Add functions for basic ai