update assignment 3