updates assignment 4