The above code looks correct, but if you look at the HTML, you’ll notice that some links don't contain a class on them. PHP will throw a WARNING because of this. Let's fix that.