HW11: AI commit: almost done!