update assignment2.