added more AI stuff