Python exercise 1 done