Added annotate.py, our main driver