Add additional code highlighting