Update method for checking of shortcode is being used

Shortcode check now works properly on index and archive pages.