Now will work if targetting a wrapping div or if targetting the img directly

Added targetting the div directly as this maybe easier to account for
in styling. Left targetting the img directly in as this can be
preferred.