AI Lab HW 1