exercism python exercise 1