fix yaml highlighting

With this fix, YAML highlighting works beautifully! Nice work :D

Fixes #3