updating assignment 4