Assignment-5 AI updated #2