Update koch.py