add argument for query.count