Assignment 4 and 5 AI