Fixing formatting & adding data-attribute selector