Updated code higlighting