Fix highlight code.
Merge pull request #8 from xtao/vilya

Fix highlight code.