AI assignment 1