Refactored plugin to not just have data- attributes