Adding support for WP_query