ai improvements