Incorporating AI in Assignments