week 6 python assignment