Incorporating AI within Assignments