Add confusion and none AI states (#188)