AI assignment 2