adding some ai stuff