Add example snippet for query function

Add example snippet for query function #255