add import in Query.java