Added some AI button functionality