updated work for assignment 3