Added check for WP version