AI2 class project