Assignment-5 AI updated