updates to Problem6