Update for WP 4.0