update assignment code