Update annotate_kos.py