forgot to add the class names to the elements. So adding that now