Exercise from TDD with Python