Better styling for code