add assignment code