Python week 1 Solution