Fix code styles

Fix styles for code blocks