Fix styles for code