update assignment5