Updating assignment 2