Improve styling, fix TOC classes