Started adding class names via js. Making plugin a tad more robust