Fix styling for code