Ai assignment code