AI assignment 3