fixes problem with highlighting
Merge pull request #14 from kris89/master

fixes problem with highlighting