update assignment 2