update assignment2