Update cite.py