fix code highlighting issues