Adding .query_by DSL to specify what keys can be queried on