added proper AI