added ai saving thing