Update kata6.py