Update 5-kittenTest.py