Put .segment class on h2 instead of div
Removed the now no longer divs and did some cleanup