Fix code block styles