Add code styling