AI algorithm improvements