updating assignment 5