Add support for multiline highlights. Closes #3

Refactored the code a bit too.