add AI, although not working...