add hw2 to AI