AI Assignment 2