add styling for some edge cases with code blocks