Python exercise 1 completed