fix code highlighting