Updating assignment 10 a bit more.