Add class for styling
Closes #12 - shortcode!