Update for a better highlight of the code