Add JS solution to detect current item, add current class