Added support for WP menus