Python TDD, finished ch4