Code highlight improvement for readme