Started work on query_index.py