updating assignment 1