Update 5_annotation.py