Clean up basic code of adding toc
Add style for toc