python lesson week 6