Add header to hw4query.py